package com.xinzhidi.common.core.redis;

import lombok.extern.slf4j.Slf4j;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.util.*;
import java.util.concurrent.TimeUnit;

/**
 * Redis工具类
 *
 * @author ZENG.XIAO.YAN
 * @date 2018年6月7日
 */
@Component
@Slf4j
public class RedisUtil {

    @Resource
    private RedisTemplate<String, Object> redisTemplate;

    /**
     * 方法描述：指定缓存失效时间
     *
     * @param key           键
     * @param expireSeconds 时间(秒)
     * @author Fei_Yu
     * @date 2021/8/20 11:23
     */
    public void setExpire(String key, long expireSeconds) {
        if (expireSeconds <= 0) {
            return;
        }
        try {
            redisTemplate.expire(key, expireSeconds, TimeUnit.SECONDS);
        } catch (Exception e) {
            log.error("指定缓存失效操作失败，错误信息{}", e.getMessage());
        }
    }

    /**
     * 方法描述：根据key 获取过期时间
     *
     * @param key 键 不能为null
     * @return java.lang.Long 时间(秒) 返回0代表为永久有效
     * @author Fei_Yu
     * @date 2021/8/20 11:27
     */
    public Long getExpire(String key) {
        return redisTemplate.getExpire(key, TimeUnit.SECONDS);
    }

    /**
     * 判断key是否存在
     *
     * @param key 键
     * @return true 存在 false不存在
     */
    public Boolean hasKey(String key) {
        try {
            return redisTemplate.hasKey(key);
        } catch (Exception e) {
            log.error("指定key判断是否存在操作错误，错误信息{}", e.getMessage());
        }
        return false;
    }


    /**
     * 方法描述：根据前缀查询keys
     *
     * @param keyPrefix keyPrefix
     * @return java.util.Collection<java.lang.String>
     * @author Fei_Yu
     * @date 2021/8/20 15:12
     */
    public Collection<String> getLikePrefixKeys(String keyPrefix) {
        return redisTemplate.keys(keyPrefix + "*");
    }

    /**
     * 普通缓存获取
     *
     * @param key 键
     * @return 值
     */
    public Object get(String key) {
        return key == null ? null : redisTemplate.opsForValue().get(key);
    }

    /**
     * 普通缓存放入
     *
     * @param key   键
     * @param value 值
     */
    public void set(String key, Object value) {
        this.setExpire(key, value, 0);
    }

    /**
     * 普通缓存放入并设置时间
     *
     * @param key           键
     * @param value         值
     * @param expireSeconds 时间(秒) time要大于0 如果time小于等于0 将设置无限期
     */
    public void setExpire(String key, Object value, long expireSeconds) {
        try {
            if (expireSeconds > 0) {
                redisTemplate.opsForValue().set(key, value, expireSeconds, TimeUnit.SECONDS);
            } else {
                redisTemplate.opsForValue().set(key, value);
            }
        } catch (Exception e) {
            log.error("放入缓存操作失败，key:{},value:{},错误信息：{}", key, value, e.getMessage());
        }
    }

    /**
     * 方法描述：递增操作
     *
     * @param key   键
     * @param delta 要增加几(大于0) 小于0则为递减
     * @return long
     * @author Fei_Yu
     * @date 2021/8/20 11:46
     */
    public long increment(String key, long delta) {
        Long increment = redisTemplate.opsForValue().increment(key, delta);
        return increment == null ? 0 : increment;
    }

    /**
     * HashGet
     *
     * @param key  键 不能为null
     * @param item 项 不能为null
     * @return 值
     */
    public Object getHashObj(String key, String item) {
        return redisTemplate.opsForHash().get(key, item);
    }

    /**
     * 获取hashKey对应的所有键值
     *
     * @param key 键
     * @return 对应的多个键值
     */
    public Map<Object, Object> getHashMap(String key) {
        return redisTemplate.opsForHash().entries(key);
    }

    /**
     * HashSet
     *
     * @param key 键
     * @param map 对应多个键值
     */
    public void addHashMap(String key, Map<String, Object> map) {
        addHashMapExpire(key, map, 0);
    }

    /**
     * HashSet 并设置时间
     *
     * @param key           键
     * @param map           对应多个键值
     * @param expireSeconds 时间(秒)
     */
    public void addHashMapExpire(String key, Map<String, Object> map, long expireSeconds) {
        redisTemplate.opsForHash().putAll(key, map);
        if (expireSeconds > 0) {
            this.setExpire(key, expireSeconds);
        }
    }

    /**
     * 向一张hash表中放入数据,如果不存在将创建
     *
     * @param key   键
     * @param item  项
     * @param value 值
     * @return true 成功 false失败
     */
    public boolean addHash(String key, String item, Object value) {
        try {
            redisTemplate.opsForHash().put(key, item, value);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 向一张hash表中放入数据,如果不存在将创建
     *
     * @param key   键
     * @param item  项
     * @param value 值
     * @param time  时间(秒) 注意:如果已存在的hash表有时间,这里将会替换原有的时间
     * @return true 成功 false失败
     */
    public boolean addHashExpire(String key, String item, Object value, long time) {
        try {
            redisTemplate.opsForHash().put(key, item, value);
            if (time > 0) {
                setExpire(key, time);
            }
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 删除hash表中的值
     *
     * @param key   键 不能为null
     * @param items 项 可以使多个 不能为null
     */
    public void delHashMap(String key, Object... items) {
        redisTemplate.opsForHash().delete(key, items);
    }

    /**
     * 判断hash表中是否有该项的值
     *
     * @param key  键 不能为null
     * @param item 项 不能为null
     * @return true 存在 false不存在
     */
    public boolean hasKeyHash(String key, String item) {
        return redisTemplate.opsForHash().hasKey(key, item);
    }

    /**
     * 方法描述：hash递增 如果不存在,就会创建一个 并把新增后的值返回
     *
     * @param key   键
     * @param item  项
     * @param delta 要增加几(大于0)
     * @return double
     * @author Fei_Yu
     * @date 2021/8/20 11:54
     */
    public double incrementHash(String key, String item, double delta) {
        return redisTemplate.opsForHash().increment(key, item, delta);
    }

    /**
     * 方法描述：根据key获取Set中的所有值
     *
     * @param key key
     * @return java.util.Set<java.lang.Object>
     * @author Fei_Yu
     * @date 2021/8/20 11:56
     */
    public Set<Object> getSet(String key) {
        try {
            return redisTemplate.opsForSet().members(key);
        } catch (Exception e) {
            e.printStackTrace();
            return new HashSet<>();
        }
    }

    /**
     * 根据value从一个set中查询,是否存在
     *
     * @param key   键
     * @param value 值
     * @return true 存在 false不存在
     */
    public boolean hasKeySet(String key, Object value) {
        try {
            Boolean member = redisTemplate.opsForSet().isMember(key, value);
            if (member != null) {
                return member;
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }

    /**
     * 将数据放入set缓存
     *
     * @param key    键
     * @param values 值 可以是多个
     * @return 成功个数
     */
    public long addSetMore(String key, Object... values) {
        return this.addSetMoreExpire(key, 0, values);
    }

    /**
     * 将set数据放入缓存
     *
     * @param key    键
     * @param time   时间(秒)
     * @param values 值 可以是多个
     * @return 成功个数
     */
    public long addSetMoreExpire(String key, long time, Object... values) {
        try {
            Long count = redisTemplate.opsForSet().add(key, values);
            if (time > 0) {
                setExpire(key, time);
            }
            return count == null ? 0 : count;
        } catch (Exception e) {
            e.printStackTrace();
            return 0;
        }
    }

    /**
     * 方法描述：获取set缓存的长度
     *
     * @param key key
     * @return long
     * @author Fei_Yu
     * @date 2021/8/20 11:57
     */
    public long getSetSize(String key) {
        try {
            Long aLong = redisTemplate.opsForSet().size(key);
            return aLong == null ? 0 : aLong;
        } catch (Exception e) {
            e.printStackTrace();
            return 0;
        }
    }

    /**
     * 移除值为value的
     *
     * @param key    键
     * @param values 值 可以是多个
     * @return 移除的个数
     */
    public long delSetMore(String key, Object... values) {
        try {
            Long count = redisTemplate.opsForSet().remove(key, values);
            return count == null ? 0 : count;
        } catch (Exception e) {
            e.printStackTrace();
            return 0;
        }
    }

    /**
     * 方法描述：获取list缓存的内容
     *
     * @param key   键
     * @param start 开始
     * @param end   结束 0 到 -1代表所有值
     * @return java.util.List<java.lang.Object>
     * @author Fei_Yu
     * @date 2021/8/20 13:50
     */
    public List<Object> getListSection(String key, long start, long end) {
        try {
            return redisTemplate.opsForList().range(key, start, end);
        } catch (Exception e) {
            e.printStackTrace();
            return new ArrayList<>();
        }
    }

    /**
     * 方法描述：获取list缓存的长度
     *
     * @param key 键
     * @return long
     * @author Fei_Yu
     * @date 2021/8/20 13:50
     */
    public long getListSize(String key) {
        try {
            Long aLong = redisTemplate.opsForList().size(key);
            return aLong == null ? 0 : aLong;
        } catch (Exception e) {
            e.printStackTrace();
            return 0;
        }
    }

    /**
     * 方法描述：通过索引 获取list中的值
     *
     * @param key   键
     * @param index 索引 index>=0时， 0 表头，1 第二个元素，依次类推；index<0时，-1，表尾，-2倒数第二个元素，依次类推
     * @return java.lang.Object
     * @author Fei_Yu
     * @date 2021/8/20 13:46
     */
    public Object getListIndex(String key, long index) {
        try {
            return redisTemplate.opsForList().index(key, index);
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }

    /**
     * 方法描述：链表list右侧添加元素
     *
     * @param key   list名字
     * @param value 值
     * @author Fei_Yu
     * @date 2022/2/21 9:58
     */
    public void addListRightPushOne(String key, Object value) {
        redisTemplate.opsForList().rightPush(key, value);
    }

    /**
     * 方法描述：获取链表list最左侧元素并移除
     *
     * @param key key
     * @return org.apache.poi.ss.formula.functions.T
     * @author Fei_Yu
     * @date 2022/2/21 10:03
     */
    public Object leftPopListOne(String key) {
        return redisTemplate.opsForList().leftPop(key);
    }

    /**
     * 方法描述：将list放入缓存
     *
     * @param key   键
     * @param value 值
     * @date 2021/8/20 13:46
     */
    public void addListOne(String key, Object value) {
        this.addListOneExpire(key, value, 0);
    }

    /**
     * 方法描述：将list放入缓存
     *
     * @param key           键
     * @param value         值
     * @param expireSeconds 时间(秒)
     * @date 2021/8/20 13:47
     */
    public void addListOneExpire(String key, Object value, long expireSeconds) {
        redisTemplate.opsForList().rightPush(key, value);
        if (expireSeconds > 0) {
            this.setExpire(key, expireSeconds);
        }
    }

    /**
     * 将list放入缓存
     *
     * @param key   键
     * @param value 值
     */
    public void addListMore(String key, List<Object> value) {
        this.addListMoreExpire(key, value, 0);
    }

    /**
     * 将list放入缓存
     *
     * @param key   键
     * @param value 值
     * @param time  时间(秒)
     */
    public void addListMoreExpire(String key, List<Object> value, long time) {
        redisTemplate.opsForList().rightPushAll(key, value);
        if (time > 0) {
            setExpire(key, time);
        }
    }

    /**
     * 根据索引修改list中的某条数据
     *
     * @param key   键
     * @param index 索引
     * @param value 值
     */
    public boolean updateListIndex(String key, long index, Object value) {
        try {
            redisTemplate.opsForList().set(key, index, value);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 移除N个值为value
     *
     * @param key       键
     * @param removeNum count > 0 : 从表头开始向表尾搜索，移除与 VALUE 相等的元素，数量为 COUNT 。
     *                  count < 0 : 从表尾开始向表头搜索，移除与 VALUE 相等的元素，数量为 COUNT 的绝对值。
     *                  count = 0 : 移除表中所有与 VALUE 相等的值。
     * @param value     值
     * @return 移除的个数
     */
    public long delListMore(String key, long removeNum, Object value) {
        try {
            Long remove = redisTemplate.opsForList().remove(key, removeNum, value);
            return remove == null ? 0 : remove;
        } catch (Exception e) {
            e.printStackTrace();
            return 0;
        }
    }

    /**
     * 方法描述：裁剪list 保留范围内的元素
     *
     * @param key        list键
     * @param startIndex 开始坐标
     * @param endIndex   结束坐标 下标 0 表示列表的第一个元素，以 1 表示列表的第二个元素，以此类推。
     *                   你也可以使用负数下标，以 -1 表示列表的最后一个元素， -2 表示列表的倒数第二个元素，以此类推。
     * @author Fei_Yu
     * @date 2022/2/21 10:40
     */
    public void trimListRange(String key, long startIndex, long endIndex) {
        redisTemplate.opsForList().trim(key, startIndex, endIndex);
    }

}
